package com.nyist.controller;

import com.alibaba.csp.sentinel.annotation.SentinelResource;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.concurrent.TimeUnit;

@RestController
public class FlowLimitController {

    @GetMapping(value = "/testA")
    public String testA(){
        return "===== TEST-A =====";
    }

    @GetMapping(value = "/testB")
    public String testB(){
        return "===== TEST-B =====";
    }


    @GetMapping(value = "/testC")
    public String testC(){
        int a=10/0;
        System.out.println("TEST-C");
        return "===== TEST-C =====";
    }

    @GetMapping(value = "/testD")
    public String testD(){
        try {
            TimeUnit.SECONDS.sleep(1);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("TEST-D");
        return "===== TEST-D =====";
    }

    //=============================热点规则

    @GetMapping("/testHotKey")
    @SentinelResource(value = "testHotKey",blockHandler = "deal_testHotKey")
    public String testHotKey(
            @RequestParam(value = "p1",required = false)String p1,
            @RequestParam(value = "p2",required = false)String p2){
        return p1+"===== testHotKey ====="+p2;
    }

    public String deal_testHotKey(String p1, String p2, BlockException exception){ // BlockException 注意别写错
        return "----deal_testHotKey---";//sentinel 默认提示 ： Blocked by Sentinel (flow limiting)
    }


}
